from rest_framework import viewsets
from .models import FocusSession, FocusLog
from .serializers import FocusSessionSerializer, FocusLogSerializer

class FocusSessionViewSet(viewsets.ModelViewSet):
    queryset = FocusSession.objects.all()
    serializer_class = FocusSessionSerializer

class FocusLogViewSet(viewsets.ModelViewSet):
    queryset = FocusLog.objects.all()
    serializer_class = FocusLogSerializer
